#!/bin/bash

# 修复 initramfs-tools 配置问题的脚本

# 颜色输出
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

print_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

print_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

print_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $1"
}

print_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

echo "======================================"
echo "🔧 修复 initramfs-tools 配置问题"
echo "======================================"

print_info "步骤1: 停止卡住的配置进程..."

# 强制终止可能卡住的进程
sudo pkill -f dpkg 2>/dev/null || true
sudo pkill -f apt 2>/dev/null || true

print_info "步骤2: 设置非交互式模式..."

# 设置环境变量避免交互提示
export DEBIAN_FRONTEND=noninteractive
export NEEDRESTART_MODE=a

print_info "步骤3: 使用默认配置完成 initramfs-tools-core 配置..."

# 方法1: 使用默认选项自动配置
echo 'initramfs-tools-core initramfs-tools-core/modules string most' | sudo debconf-set-selections
echo 'initramfs-tools-core initramfs-tools-core/compress string gzip' | sudo debconf-set-selections

# 强制配置包，使用默认选项
sudo dpkg --configure -a --force-confdef --force-confold

if [ $? -eq 0 ]; then
    print_success "initramfs-tools 配置问题已修复"
else
    print_warning "方法1失败，尝试方法2..."
    
    # 方法2: 强制重新安装
    print_info "步骤4: 强制重新安装 initramfs-tools..."
    sudo apt-get remove --purge initramfs-tools initramfs-tools-core -y
    sudo apt-get autoremove -y
    sudo apt-get update
    sudo apt-get install initramfs-tools initramfs-tools-core -y
    
    if [ $? -eq 0 ]; then
        print_success "重新安装成功"
    else
        print_error "重新安装也失败，尝试方法3..."
        
        # 方法3: 手动清理并重试
        print_info "步骤5: 手动清理并重试..."
        sudo rm -f /var/lib/dpkg/info/initramfs-tools-core.postinst
        sudo rm -f /var/lib/dpkg/info/initramfs-tools.postinst
        sudo dpkg --remove --force-remove-reinstreq initramfs-tools-core
        sudo dpkg --remove --force-remove-reinstreq initramfs-tools
        sudo apt-get update
        sudo apt-get install initramfs-tools -y
        
        if [ $? -eq 0 ]; then
            print_success "手动修复成功"
        else
            print_error "所有方法都失败，需要手动处理"
            exit 1
        fi
    fi
fi

print_info "步骤6: 安装 npm..."

# 现在安装 npm
sudo apt-get update
sudo apt-get install npm -y

if [ $? -eq 0 ]; then
    print_success "npm 安装成功"
    npm --version
else
    print_error "npm 安装失败"
    exit 1
fi

print_info "步骤7: 清理系统..."

# 清理
sudo apt-get autoremove -y
sudo apt-get autoclean

print_success "✅ 系统修复完成！现在可以重新运行安装脚本"

echo ""
print_info "建议的下一步操作："
echo "1. 重新运行一键安装脚本"
echo "2. 或者手动执行安装步骤"
